cd ../../../
exp_id="cdn"
dataset='hico'
dataset_path='data/hico_20160224_det'
addr="127.0.0.1"
port=1119
num_gpu=4
pretrained='/root/whzh/codes/HOI/cdn/params/Detr/detr-r50-pre-2stage-q64.pth'
output_dir=logs/$dataset/$exp_id/stage1/
#'params/ConditionalDetr/ConditionalDETR_r50-pre-2stage.pth'
#'pretrained/hico/hico_cdn_s.pth'

export OMP_NUM_THREADS=1
python main.py \
--world_size $num_gpu \
--master_addr $addr \
--master_port $port \
--exp_id $exp_id \
--pretrained $pretrained \
--output_dir $output_dir \
--dataset_file $dataset \
--hoi_path $dataset_path \
--num_obj_classes 80 \
--num_verb_classes 117 \
--dec_layers_hopd 3 \
--dec_layers_interaction 3 \
--batch_size 4 \
--lr 1e-4 \
--epochs 90 \
--lr_drop 60 \
--num_queries 64 \
--dim_feedforward 2048 \
--set_cost_verb_class 2 \
--verb_loss_coef 2 \
--num_feature_levels 1 \
--use_nms_filter


# python main.py \
# --world_size $num_gpu \
# --master_addr $addr \
# --master_port $port \
# --pretrained logs/$dataset/$exp_id/stage1/checkpoint_best.pth \
# --output_dir logs/$dataset/$exp_id/stage2/ \
# --dataset_file $dataset \
# --hoi_path $dataset_path \
# --num_obj_classes 80 \
# --num_verb_classes 117 \
# --backbone resnet50 \
# --num_queries 64 \
# --dec_layers_hopd 3 \
# --dec_layers_interaction 3 \
# --epochs 10 \
# --freeze_mode 1 \
# --obj_reweight \
# --verb_reweight \
# --queue_size 9408 \
# --p_obj 0.7 \
# --p_verb 0.7 \
# --lr 5e-6 \
# --lr_backbone 6e-7 \
# --use_nms_filter